from histogram import HistogramCollection, Histogram1D, Histogram2D

hcoll = HistogramCollection(
    [Histogram2D(name='m_kpi_vs_m_pik',
                 nbins=(250,250),
                 aranges=((1.75, 2.0), (1.75, 2.0)),
                 xlabel=r'M($K^+\pi^-$)',
                 xunits=r'GeV',
                 ylabel=r'M($K^-\pi^+$)',
                 yunits=r'GeV',
                 assoc_cols=('m_kpi', 'm_pik'),
                 ),
     Histogram1D(name='m_kpi_wide',
                 nbins=1000,
                 arange=(0.6, 5.6),
                 xlabel=r'M($K^+\pi^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_kpi',
                 ),
     Histogram1D(name='m_pik_wide',
                 nbins=1000,
                 arange=(0.6, 5.6),
                 xlabel=r'M($K^-\pi^+$)',
                 xunits=r'GeV',
                 assoc_cols='m_pik',
                 ),
     Histogram1D(name='m_kpi',
                 nbins=250,
                 arange=(1.75, 2.0),
                 xlabel=r'M($K^+\pi^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_kpi',
                 ),
     Histogram1D(name='m_pik',
                 nbins=250,
                 arange=(1.75, 2.0),
                 xlabel=r'M($K^-\pi^+$)',
                 xunits=r'GeV',
                 assoc_cols='m_pik',
                 ),
     Histogram1D(name='m_pik_fit',
                 nbins=150,
                 arange=(1.8, 1.95),
                 xlabel=r'M($K^-\pi^+$)',
                 xunits=r'GeV',
                 assoc_cols='m_pik',
                 ),
     Histogram1D(name='m_pipi_wide',
                 nbins=1000,
                 arange=(0.3, 5.3),
                 xlabel=r'M($\pi^+\pi^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_pipi',
                 ),
     Histogram1D(name='m_kk_wide',
                 nbins=1000,
                 arange=(1.0, 6.0),
                 xlabel=r'M($K^+K^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_kk',
                 ),
     Histogram1D(name='m_pipi',
                 nbins=250,
                 arange=(1.75, 2.0),
                 xlabel=r'M($\pi^+\pi^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_pipi',
                 ),
     Histogram1D(name='m_kk',
                 nbins=250,
                 arange=(1.75, 2.0),
                 xlabel=r'M($K^+K^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_kk',
                 ),
     ])

